package com.example.template;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.PutObjectRequest;
import com.example.properties.OssProperties;

import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

/**
 * 自动装配注入进Spring Boot操作的类
 */
public class OssTemplate {
    private OssProperties ossProperties;

    public OssTemplate(OssProperties ossProperties) {
        this.ossProperties = ossProperties;
    }

    // 测试是否注入成功
    public void test() {
        System.out.println("注入成功，bucket名称为：" + ossProperties.getBucketName());
    }

    public String upload(String filename, InputStream is) {
        // yourEndpoint填写Bucket所在地域对应的Endpoint。
        // 以华北2（北京）为例，Endpoint填写为https://oss-cn-beijing.aliyuncs.com（不加https://也可以）
        String endpoint = ossProperties.getEndpoint();
        // 阿里云主账号AccessKey拥有所有API的访问权限，风险很高。
        // 强烈建议您创建并使用RAM账号进行API访问或日常运维，请登录 https://ram.console.aliyun.com 创建RAM账号。
        String accessKeyId = ossProperties.getAccessKey();
        String accessKeySecret = ossProperties.getSecret();

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);

        // 存储对象的名称：加了/，则存储的时候就产生了文件夹
        String storeName = new SimpleDateFormat("yyyy/MM/dd").format(new Date())
                + "/"
                + UUID.randomUUID()
                + filename.substring(filename.lastIndexOf("."));
        System.out.println(storeName);
        // 依次填写Bucket名称（例如examplebucket）和Object完整路径（例如exampledir/exampleobject.txt）。
        // Object完整路径中不能包含Bucket名称。
        // ossClient.putObject(ossProperties.getBucketName(), storePath, is);
        // 或者可写成
        PutObjectRequest request = new PutObjectRequest(ossProperties.getBucketName(), storeName, is);
        ossClient.putObject(request);


        // 关闭OSSClient。
        ossClient.shutdown();

        // 阿里云返回的url
        String url = "https://" + ossProperties.getBucketName() + "." +
                ossProperties.getEndpoint().substring(ossProperties.getEndpoint().lastIndexOf("/") + 1) + "/" + storeName;
        return url;
    }


    public void remove(String fileUrl) {
        // yourEndpoint填写Bucket所在地域对应的Endpoint。
        // 以华东1（杭州）为例，Endpoint填写为https://oss-cn-hangzhou.aliyuncs.com。
        String endpoint = ossProperties.getEndpoint();
        // 阿里云账号AccessKey拥有所有API的访问权限，风险很高。
        // 强烈建议您创建并使用RAM用户进行API访问或日常运维，请登录RAM控制台创建RAM用户。
        String accessKeyId = ossProperties.getAccessKey();
        String accessKeySecret = ossProperties.getSecret();

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);

        // 填写Bucket名称。
        String bucketName = ossProperties.getBucketName();
        // 填写文件完整路径。文件完整路径中不能包含Bucket名称。
        // 2022/01/21/f0870eb3-4714-4fae-9fc3-35e72202f193.jpg

        // 删除文件或目录。如果要删除目录，目录必须为空。
        ossClient.deleteObject(bucketName, fileUrl);

        // 关闭OSSClient。
        ossClient.shutdown();
    }
}
